from rest_framework import viewsets
from .models import Host, City, DataCenter, HostStatistics
from .serializers import HostSerializer, CitySerializer, DataCenterSerializer, HostStatisticsSerializer

class HostViewSet(viewsets.ModelViewSet):
    queryset = Host.objects.all()
    serializer_class = HostSerializer

class CityViewSet(viewsets.ModelViewSet):
    queryset = City.objects.all()
    serializer_class = CitySerializer

class DataCenterViewSet(viewsets.ModelViewSet):
    queryset = DataCenter.objects.all()
    serializer_class = DataCenterSerializer

class HostStatisticsViewSet(viewsets.ModelViewSet):
    queryset = HostStatistics.objects.all()
    serializer_class = HostStatisticsSerializer
